Invalid instance which ends at line: 9614 has been dropped.
Invalid instance which ends at line: 121 has been dropped.
Invalid instance which ends at line: 216 has been dropped.
Invalid instance which ends at line: 121 has been dropped.
Invalid instance which ends at line: 216 has been dropped.
Start constructing character vocabulary.
In total, there are 60 distinct characters.
Found 224 out of 437 words in the pre-training embedding.
In total 3 datasets:
	train has 1366 instances.
	dev has 405 instances.
	test has 405 instances.
In total 2 vocabs:
	words has 437 entries.
	target has 13 entries.

input fields after batch(if batch size is 2):
	target: (1)type:torch.Tensor (2)dtype:torch.int64, (3)shape:torch.Size([2, 3]) 
	words: (1)type:torch.Tensor (2)dtype:torch.int64, (3)shape:torch.Size([2, 3]) 
	seq_len: (1)type:torch.Tensor (2)dtype:torch.int64, (3)shape:torch.Size([2]) 
target fields after batch(if batch size is 2):
	target: (1)type:torch.Tensor (2)dtype:torch.int64, (3)shape:torch.Size([2, 3]) 
	seq_len: (1)type:torch.Tensor (2)dtype:torch.int64, (3)shape:torch.Size([2]) 

Preds: 
tensor([[9, 4, 0, 0, 0, 0, 4],
        [9, 4, 0, 0, 0, 0, 4]], device='cuda:0')
Target: 
tensor([[8, 6, 6, 0, 0, 0, 0],
        [8, 6, 6, 0, 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[9, 4, 0, 0, 0, 0, 4],
        [9, 4, 0, 0, 0, 4, 0]], device='cuda:0')
Target: 
tensor([[8, 6, 6, 0, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0]], device='cuda:0')
training epochs started 2022-08-11-22-00-29-283793
Preds: 
tensor([[ 1,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 4,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 2, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [1, 0, 0,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  4,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  4,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,
          4,  4,  4,  4,  4,  4,  4,  4,  4,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 6, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4,  0,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 8, 6, 6, 0, 4, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 8, 6, 6, 0, 4, 0, 0, 0, 0],
        [1, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [1, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.25 seconds!
Evaluation on dev at Epoch 1/15. Step:43/645: 
EvaluateNER: acc=16.691358024691358

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  9, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          9, 12,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 0,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  1,  0,  0,  0,  0,  7,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  1,  0,  0,  0,  0,  7,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  1,  2,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  1,  2,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  3,  5,  0,  4, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10,
         10, 10, 10, 10, 10, 10, 10, 10, 10,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,
          5,  5,  5,  5,  5,  5,  5,  5,  5,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  3,  5,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  3,  5,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 6, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 6, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.28 seconds!
Evaluation on dev at Epoch 2/15. Step:86/645: 
EvaluateNER: acc=16.390123456790125

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  9, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          9, 12,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 0,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  8,  6,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  8,  6,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  1,  2,  0,  0,  8,  6,  6,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  1,  2,  0,  0,  8,  6,  6,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  3,  5,  0,  8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,
          6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 1,  3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 6, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 6, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.3 seconds!
Evaluation on dev at Epoch 3/15. Step:129/645: 
EvaluateNER: acc=16.65432098765432

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  9, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          9, 12,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9,  3,  5,  0,  8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,
          6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9,  3,  5,  0,  0,  1,  0,  0,  0,  0,  0,  0,  0],
        [ 9,  3,  5,  0,  0,  1,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 8, 6, 6, 0, 4, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 8, 6, 6, 0, 4, 0, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.33 seconds!
Evaluation on dev at Epoch 4/15. Step:172/645: 
EvaluateNER: acc=16.56543209876543

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  1,  0,  0,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  1,  0,  0,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  3,  5,  0,  8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,
          6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,
          5,  5,  5,  5,  5,  5,  5,  5,  5,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 1,  3,  5,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 4, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 4, 0, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.28 seconds!
Evaluation on dev at Epoch 5/15. Step:215/645: 
EvaluateNER: acc=16.496296296296297

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  9, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          9, 12,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  1,  0,  0,  0,  1,  2,  0,  0,  9, 12,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  1,  0,  0,  0,  1,  2,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  3,  5,  0,  8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,
          6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,
          5,  5,  5,  5,  5,  5,  5,  5,  5,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 6, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 6, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.32 seconds!
Evaluation on dev at Epoch 6/15. Step:258/645: 
EvaluateNER: acc=16.533333333333335

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  9, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12,
         12, 12, 12, 12, 12, 12, 12, 12, 12,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,
          5,  5,  5,  5,  5,  5,  5,  5,  5,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 6, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 6, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.26 seconds!
Evaluation on dev at Epoch 7/15. Step:301/645: 
EvaluateNER: acc=16.649382716049384

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  3,  5,  0,  9, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12,
         12, 12, 12, 12, 12, 12, 12, 12, 12,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,
          5,  5,  5,  5,  5,  5,  5,  5,  5,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.25 seconds!
Evaluation on dev at Epoch 8/15. Step:344/645: 
EvaluateNER: acc=16.49135802469136

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  1,  2,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  1,  2,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  3,  5,  0,  9, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12,
         12, 12, 12, 12, 12, 12, 12, 12, 12,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,
          5,  5,  5,  5,  5,  5,  5,  5,  5,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 1,  3,  5,  0,  0,  1,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  1,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.25 seconds!
Evaluation on dev at Epoch 9/15. Step:387/645: 
EvaluateNER: acc=16.528395061728396

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  3,  5,  0,  9, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12,
         12, 12, 12, 12, 12, 12, 12, 12, 12,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,
          5,  5,  5,  5,  5,  5,  5,  5,  5,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.25 seconds!
Evaluation on dev at Epoch 10/15. Step:430/645: 
EvaluateNER: acc=16.54320987654321

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  1,  2,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  1,  2,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  3,  5,  0,  8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,
          6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,
          5,  5,  5,  5,  5,  5,  5,  5,  5,  0],
        [ 0,  4,  0,  0,  0,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  8,  6,  6,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 4, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 4, 0, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.25 seconds!
Evaluation on dev at Epoch 11/15. Step:473/645: 
EvaluateNER: acc=16.54320987654321

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  1,  0,  0,  0,  1,  2,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  1,  0,  0,  0,  1,  2,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  3,  5,  0,  8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,
          6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,
          5,  5,  5,  5,  5,  5,  5,  5,  5,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.26 seconds!
Evaluation on dev at Epoch 12/15. Step:516/645: 
EvaluateNER: acc=16.580246913580247

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  1,  0,  1,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  1,  0,  1,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  1,  2,  0,  8,  6,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  3,  5,  0,  9, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12,
         12, 12, 12, 12, 12, 12, 12, 12, 12,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,
          5,  5,  5,  5,  5,  5,  5,  5,  5,  0],
        [ 0,  4,  0,  0,  0,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 9, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 1,  3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  8,  6,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 6, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 6, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.28 seconds!
Evaluation on dev at Epoch 13/15. Step:559/645: 
EvaluateNER: acc=16.506172839506174

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  1,  0,  0,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  1,  0,  0,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  3,  5,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,
          5,  5,  5,  5,  5,  5,  5,  5,  5,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.3 seconds!
Evaluation on dev at Epoch 14/15. Step:602/645: 
EvaluateNER: acc=16.585185185185185

Preds: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0,  3,  5,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          3,  5,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 7,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 1, 0,  ..., 0, 0, 0],
        [1, 2, 0,  ..., 0, 0, 0],
        [7, 0, 9,  ..., 0, 0, 0],
        ...,
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0],
        [0, 0, 0,  ..., 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  9, 12,  0,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  1,  0,  1,  0,  1,  2,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  1,  0,  1,  0,  1,  2,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 0,  0,  0,  0,  1,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 9, 12,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
          9,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  3,  5,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,  5,
          5,  5,  5,  5,  5,  5,  5,  5,  5,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0],
        [8, 6, 6, 0, 9, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  9,  9,  9,  9,  9,  9,  9,  9,  0],
        [ 8,  6,  6,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  8,  6,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12, 12, 12, 12, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 9, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 9, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [0, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.26 seconds!
Evaluation on dev at Epoch 15/15. Step:645/645: 
EvaluateNER: acc=16.582716049382714

Reloaded the best model.

In Epoch:1/Step:43, got best dev performance:
EvaluateNER: acc=16.691358024691358
Preds: 
tensor([[ 1,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 1,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 1,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 4,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 4,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  4,
          0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0]], device='cuda:0')
Preds: 
tensor([[3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 4, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]],
       device='cuda:0')
Target: 
tensor([[8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 0, 1, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 4, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 3, 5, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 5, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]],
       device='cuda:0')
Preds: 
tensor([[3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 0]],
       device='cuda:0')
Target: 
tensor([[3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 0, 1, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  1,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 8,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  0],
        [ 1,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  0],
        [ 4,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Target: 
tensor([[ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 0, 0, 0, 1, 2, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0],
        [ 7,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  9,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  7,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[8, 6, 0, 7, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 7, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 7, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 7, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 7, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 7, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 5, 5, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
         0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0],
        [ 0,  0,  7,  0,  9, 12,  0,  7,  9, 12,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  5,  0,  4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]],
       device='cuda:0')
Target: 
tensor([[ 8,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  0,  7, 11, 11,  0,  0,  0,  9, 12,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10, 10,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  4,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 7,  8,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 4,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  1,  2,  0,  0,  0,  4,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Target: 
tensor([[ 4,  0,  0,  0,  1,  2,  0,  9,  0,  9,  0,  9, 12,  0,  0,  9, 12, 12,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0],
        [ 8,  6,  6,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  1,  0,  8,  6,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  4,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0],
        [ 0,  0,  0,  0,  1,  2,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[ 1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,
          4,  4,  4,  4,  4,  4,  4,  4,  4,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  9, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  1,  2,  0,  1,  2,  2,  0,  0,  0,  0,  0,  0,  0,
          1,  2,  0,  8,  6,  6,  0,  0,  7,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  4,  0,  0,  0,  0,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  4,  7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
          0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[ 1,  2,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 9, 12, 12,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]],
       device='cuda:0')
Preds: 
tensor([[1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [8, 6, 6, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [3, 5, 0, 4, 0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0],
        [8, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 4, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 9, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 1, 2, 0, 9, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 9, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[8, 6, 6, 0, 1, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0],
        [8, 6, 6, 0, 1, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0],
        [8, 6, 6, 0, 1, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0],
        [8, 6, 6, 0, 1, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0],
        [8, 6, 6, 0, 1, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0],
        [8, 6, 6, 0, 1, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0],
        [8, 6, 6, 0, 1, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[0, 1, 2, 0, 9, 0, 0, 0],
        [1, 2, 0, 0, 0, 0, 0, 0],
        [1, 2, 0, 0, 9, 0, 0, 0],
        [0, 1, 2, 0, 9, 0, 0, 0],
        [1, 2, 0, 0, 9, 0, 0, 0],
        [0, 1, 2, 0, 9, 0, 0, 0],
        [0, 0, 0, 9, 0, 9, 0, 0],
        [1, 2, 0, 0, 9, 0, 0, 0],
        [0, 1, 2, 0, 9, 0, 0, 0],
        [1, 2, 0, 0, 9, 0, 0, 0],
        [0, 1, 2, 0, 9, 0, 0, 0],
        [1, 2, 0, 0, 9, 0, 0, 0],
        [0, 1, 2, 0, 9, 0, 0, 0],
        [1, 2, 0, 0, 9, 0, 0, 0],
        [0, 1, 2, 0, 9, 0, 0, 0],
        [1, 2, 0, 0, 9, 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 8,  6,  6,  0,  1,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  0,  0,  9,  0,  9,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 0,  1,  2,  0,  9,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  4,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 4, 10,  1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  0,  8,  6,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  1,  2,  2,  2,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7,  9,  0,  9, 12,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  0,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4,  0,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Target: 
tensor([[ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  5,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  0,  1,  0,  1,  2,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  0,  4, 10,  1,  2,  0,  0,  0,  0],
        [ 1,  2,  0,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0],
        [ 3,  5,  0,  7, 11,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 7, 11,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0],
        [ 1,  2,  2,  0,  0,  0,  0,  8,  6,  6,  0,  0,  0],
        [ 4,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0]], device='cuda:0')
Preds: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 8, 6, 6, 0, 4, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Preds: 
tensor([[1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 8, 6, 6, 0, 4, 0, 0, 0, 0],
        [1, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [1, 0, 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0],
        [4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Target: 
tensor([[1, 2, 2, 0, 0, 0, 0, 8, 6, 6, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 1, 2, 0, 0, 8, 6, 6, 0, 8, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], device='cuda:0')
Evaluate data in 0.33 seconds!
[tester] 
EvaluateNER: acc=14.602469135802469
